Fixed the bug that broke it