Fix bug not compiling