Bug fixes (omg, fucking fixes)