Removed register
Fixed little bug