fixed a little bug i hope