Fixes so it compiles