Small fuckup fix