Fixing stuff

That I forgot