Fix for error on build