Quick fix - fixes #5.