Fix some weird import