Small fix for end