Fixed a bug, thx lars