Fix bug to make code runnable again