Bug fix... :(