Fixing some build issues with missing code

- Correcting syntax error
- Adding in test appropriately