Fix the build - thanks @jbpros!