Minor change and it works!