I updated the code and did some refactoring. This bug should be fixed now.