Small fix

Removing beta testing code