Small bug fixed with naming.