Fixed a bit of stuff