New photo box classes broke extension, this fixes that