Broken test fixed