Fixed Class not found error on build